package com.whacos.common.utils;

/**
 * @author xiongdun
 * @description 功能：根据文件名称判断类型
 * 接受参数类型：String
 * 返回参数类型：String
 * 备注：文件类型不完善，有需要的自行添加
 * @since 2018/11/17 17:11
 */
public class FileType {

    public static int fileType(String fileName) {
        if (fileName == null) {
            fileName = "文件名为空！";
            return 500;
        } else {
            // 获取文件后缀名并转化为写，用于后续比较
            String fileType = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()).toLowerCase();
            // 创建图片类型数组0
            String[] images = {"bmp", "jpg", "jpeg", "png", "tiff", "gif", "pcx", "tga", "exif", "fpx", "svg", "psd",
                    "cdr", "pcd", "dxf", "ufo", "eps", "ai", "raw", "wmf"};

            for (String image : images) {
                if (image.equals(fileType)) {
                    return 0;
                }
            }

            // 创建文档类型数组1
            String[] documents = {"txt", "doc", "docx", "xls", "htm", "html", "jsp", "rtf", "wpd", "pdf", "ppt"};

            for (String document : documents) {
                if (document.equals(fileType)) {
                    return 1;
                }
            }

            // 创建视频类型数组2
            String[] videos = {"mp4", "avi", "mov", "wmv", "asf", "navi", "3gp", "mkv", "f4v", "rmvb", "webm"};

            for (String video : videos) {
                if (video.equals(fileType)) {
                    return 2;
                }
            }

            // 创建音乐类型数组3
            String[] musics = {"mp3", "wma", "wav", "mod", "ra", "cd", "md", "asf", "aac", "vqf", "ape", "mid", "ogg",
                    "m4a", "vqf"};

            for (String music : musics) {
                if (music.equals(fileType)) {
                    return 3;
                }
            }
        }
        //4
        return 99;
    }
}